#ifndef __IANGELS_MODEL_VIT_H__
#define __IANGELS_MODEL_VIT_H__

#include <memory>

#include "ugraph/graph.h"
#include "status.h"
#include "config.h"

namespace iangels {

class ViT {
public:
    ViT(const Config& config): config_(config) {}
    ~ViT() {}

    Ret build_graph();

private:
    uGraph graph_; // graph for the model
    const Config& config_;
}; // class ViT

} // namespace iangels

#endif // __IANGELS_MODEL_VIT_H__